/**
 * 
 */
package com.zte.claa.inficombo.csa.app.util;

import java.awt.Dimension;
import java.awt.Graphics2D;
import java.awt.Image;
import java.awt.Rectangle;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

import javax.imageio.ImageIO;
import javax.imageio.stream.FileImageInputStream;
import javax.imageio.stream.FileImageOutputStream;

/**
 * @author zhanghao
 *
 */
public class ImageUtils {
	
	private static Map<String, Byte> IMAGETYPE_MAP = new HashMap<>();
	private static Map<Byte, String> CONTENTTYPE_MAP = new HashMap<>();
	private static Map<String, String> IMAGE_MAP = new HashMap<>();
	
	public static final String DEFAULT_IMAGE_CONTENTTYPE = "image/png";
	public static final String DEFAULT_IMAGE_TYPE = "jpg";
	
	static{
		IMAGETYPE_MAP.put("png", (byte)0);
		IMAGETYPE_MAP.put("jpeg", (byte)1);
		IMAGETYPE_MAP.put("jpg", (byte)2);
		IMAGETYPE_MAP.put("gif", (byte)3);
		
		CONTENTTYPE_MAP.put((byte)0, "image/png");
		CONTENTTYPE_MAP.put((byte)1, "image/jpeg");
		CONTENTTYPE_MAP.put((byte)2, "image/jpeg");
		CONTENTTYPE_MAP.put((byte)3, "image/gif");
		
		IMAGE_MAP.put("png", "image/png");
		IMAGE_MAP.put("jpeg","image/jpeg");
		IMAGE_MAP.put("jpg", "image/jpeg");
		IMAGE_MAP.put("gif", "image/gif");
	}
	
	// 根据图像的后缀，获取图片的类型
	public static byte getImageType(String suffix) throws Exception{
		Byte imageType = IMAGETYPE_MAP.get(suffix.toLowerCase());
		if (imageType == null){
			throw new Exception("不支持的图片类型");
		}
		return imageType.byteValue();
	}
	
	// 根据图片的类型，返回图片的contenttype
	public static String getImageContentType(byte imagetype) throws Exception{
		String contentType = CONTENTTYPE_MAP.get(imagetype);
		if (contentType == null){
			throw new Exception("不支持的图片类型");
		}
		return contentType;
	}
	
	// 根据图片的类型，返回图片的contenttype
	public static String getImageContentType(byte imagetype, String defaltValue){
		String contentType = CONTENTTYPE_MAP.get(imagetype);
		if (contentType == null){
			return defaltValue;
		}
		return contentType;
	}
	
	// 根据图片的类型，返回图片的contenttype
	public static String getImageContentType(String imagetype, String defaltValue){
		String contentType = IMAGE_MAP.get(imagetype);
		if (contentType == null){
			return defaltValue;
		}
		return contentType;
	}
	
	  /**
	   * 对图片进行旋转
	   *
	   * @param src  被旋转图片二进制码流
	   * @param angel 旋转角度
	   * @return 旋转后的图片
	 * @throws IOException 
	   */
	   public static byte[] Rotate(byte[] src, double angel, String imgfmt) throws IOException {
		   if(null == src || src.length == 0){
			   return null;
		   }
		   byte[] outbytes = null;
		   InputStream buffin = new ByteArrayInputStream(src);
           BufferedImage imagesrc = ImageIO.read(buffin);
           BufferedImage imagedst = ImageUtils.Rotate(imagesrc, angel);
           outbytes = ImageUtils.image2byte(imagedst,imgfmt);
		   return outbytes;
	   }
	
	  /**
	   * 对图片进行旋转
	   *
	   * @param src  被旋转图片
	   * @param angel 旋转角度
	   * @return 旋转后的图片
	   */
	  public static BufferedImage Rotate(Image src, double angel) {
	    int src_width = src.getWidth(null);
	    int src_height = src.getHeight(null);
	    // 计算旋转后图片的尺寸
	    Rectangle rect_des = CalcRotatedSize(new Rectangle(new Dimension(
	        src_width, src_height)), angel);
	    BufferedImage res = null;
	    res = new BufferedImage(rect_des.width, rect_des.height,
	        BufferedImage.TYPE_INT_RGB);
	    Graphics2D g2 = res.createGraphics();
	    // 进行转换
	    g2.translate((rect_des.width - src_width) / 2,
	        (rect_des.height - src_height) / 2);
	    g2.rotate(Math.toRadians(angel), src_width / 2, src_height / 2);
	  
	    g2.drawImage(src, null, null);
	    return res;
	  }
  
	  /**
	   * 计算旋转后的图片
	   *
	   * @param src  被旋转的图片
	   * @param angel 旋转角度
	   * @return 旋转后的图片
	   */
	  public static Rectangle CalcRotatedSize(Rectangle src, double angel) {
	    // 如果旋转的角度大于90度做相应的转换
	    if (angel >= 90) {
	      if (angel / 90 % 2 == 1) {
	        int temp = src.height;
	        src.height = src.width;
	        src.width = temp;
	      }
	      angel = angel % 90;
	    }
	  
	    double r = Math.sqrt(src.height * src.height + src.width * src.width) / 2;
	    double len = 2 * Math.sin(Math.toRadians(angel) / 2) * r;
	    double angel_alpha = (Math.PI - Math.toRadians(angel)) / 2;
	    double angel_dalta_width = Math.atan((double) src.height / src.width);
	    double angel_dalta_height = Math.atan((double) src.width / src.height);
	  
	    int len_dalta_width = (int) (len * Math.cos(Math.PI - angel_alpha
	        - angel_dalta_width));
	    int len_dalta_height = (int) (len * Math.cos(Math.PI - angel_alpha
	        - angel_dalta_height));
	    int des_width = src.width + len_dalta_width * 2;
	    int des_height = src.height + len_dalta_height * 2;
	    return new Rectangle(new Dimension(des_width, des_height));
	  }
	  

	  //BufferedImage转流
	  public static ByteArrayOutputStream image2Stream(BufferedImage image, String imgfmt) {
	        ByteArrayOutputStream out = new ByteArrayOutputStream();
	        try {
	            ImageIO.write(image, imgfmt, out);
	            out.flush();
	        } catch (IOException e) {
	            e.printStackTrace();
	        } finally {
	            try {
	                out.close();
	            } catch (IOException e) {
	                e.printStackTrace();
	            }
	        }
	        return out;
	  }
	    
	  //BufferedImage转byte[]
	  public static byte[] image2byte(BufferedImage image, String imgfmt) {
		  if(null == image || null == imgfmt || imgfmt.equals("")){
			  return null;
		  }
	      ByteArrayOutputStream out = image2Stream(image, imgfmt);
	      byte[] outToByteArray = out.toByteArray();
	      return outToByteArray;
	  }

}
